/*
 * Copyright (C) 2013 Square, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.squareup.picasso;

import android.annotation.SuppressLint;
import android.annotation.TargetApi;
import android.content.ContentResolver;
import android.content.Context;
import android.content.UriMatcher;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.net.Uri;
import android.provider.ContactsContract;

import java.io.IOException;
import java.io.InputStream;

import static android.os.Build.VERSION.SDK_INT;
import static android.os.Build.VERSION_CODES.ICE_CREAM_SANDWICH;
import static android.provider.ContactsContract.Contacts.openContactPhotoInputStream;
import static com.squareup.picasso.Picasso.LoadedFrom.DISK;

@SuppressLint("NewApi")
class ContactsPhotoBitmapHunter extends BitmapHunter {
	/**
	 * A lookup uri (e.g.
	 * content://com.android.contacts/contacts/lookup/3570i61d948d30808e537)
	 */
	private static final int ID_LOOKUP = 1;
	/**
	 * A contact thumbnail uri (e.g.
	 * content://com.android.contacts/contacts/38/photo)
	 */
	private static final int ID_THUMBNAIL = 2;
	/** A contact uri (e.g. content://com.android.contacts/contacts/38) */
	private static final int ID_CONTACT = 3;
	/**
	 * A contact display photo (high resolution) uri (e.g.
	 * content://com.android.contacts/display_photo/5)
	 */
	private static final int ID_DISPLAY_PHOTO = 4;

	private static final UriMatcher matcher;

	static {
		matcher = new UriMatcher(UriMatcher.NO_MATCH);
		matcher.addURI(ContactsContract.AUTHORITY, "contacts/lookup/*/#",
				ID_LOOKUP);
		matcher.addURI(ContactsContract.AUTHORITY, "contacts/lookup/*",
				ID_LOOKUP);
		matcher.addURI(ContactsContract.AUTHORITY, "contacts/#/photo",
				ID_THUMBNAIL);
		matcher.addURI(ContactsContract.AUTHORITY, "contacts/#", ID_CONTACT);
		matcher.addURI(ContactsContract.AUTHORITY, "display_photo/#",
				ID_DISPLAY_PHOTO);
	}

	final Context context;

	@SuppressWarnings("rawtypes")
	ContactsPhotoBitmapHunter(Context context, Picasso picasso,
			Dispatcher dispatcher, Cache cache, Stats stats, Action action) {
		super(picasso, dispatcher, cache, stats, action);
		this.context = context;
	}

	@Override
	Bitmap decode(Request data) throws IOException {
		InputStream is = null;
		try {
			is = getInputStream();
			return decodeStream(is, data);
		} finally {
			Utils.closeQuietly(is);
		}
	}

	@Override
	Picasso.LoadedFrom getLoadedFrom() {
		return DISK;
	}

	private InputStream getInputStream() throws IOException {
		ContentResolver contentResolver = context.getContentResolver();
		Uri uri = getData().uri;
		switch (matcher.match(uri)) {
		case ID_LOOKUP:
			uri = ContactsContract.Contacts.lookupContact(contentResolver, uri);
			if (uri == null) {
				return null;
			}
			// Resolved the uri to a contact uri, intentionally fall through to
			// process the resolved uri
		case ID_CONTACT:
			if (SDK_INT < ICE_CREAM_SANDWICH) {
				return openContactPhotoInputStream(contentResolver, uri);
			} else {
				return ContactPhotoStreamIcs.get(contentResolver, uri);
			}
		case ID_THUMBNAIL:
		case ID_DISPLAY_PHOTO:
			return contentResolver.openInputStream(uri);
		default:
			throw new IllegalStateException("Invalid uri: " + uri);
		}
	}

	private Bitmap decodeStream(InputStream stream, Request data)
			throws IOException {
		if (stream == null) {
			return null;
		}
		final BitmapFactory.Options options = createBitmapOptions(data);
		if (requiresInSampleSize(options)) {
			InputStream is = getInputStream();
			try {
				BitmapFactory.decodeStream(is, null, options);
			} finally {
				Utils.closeQuietly(is);
			}
			calculateInSampleSize(data.targetWidth, data.targetHeight, options);
		}
		return BitmapFactory.decodeStream(stream, null, options);
	}

	@TargetApi(ICE_CREAM_SANDWICH)
	private static class ContactPhotoStreamIcs {
		static InputStream get(ContentResolver contentResolver, Uri uri) {
			return openContactPhotoInputStream(contentResolver, uri, true);
		}
	}
}
